package common.utils;

import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import org.apache.commons.lang3.StringUtils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 表格分页数据对象
 *
 * @author song
 */

@Data
public class RTableData<T> implements Serializable {
    private static final long serialVersionUID = 1L;

    @ApiModelProperty("当前列表数据数组")
    private T list;

    @ApiModelProperty("列表数据总数")
    private int total;

    /**
     * 表格数据对象
     */
    public RTableData() {

    }

    /**
     * 分页
     *
     * @param t     列表数据对象
     * @param total 总记录数
     */
    public RTableData(T t, int total) {
        this.list = t;
        this.total = total;
    }

    /**
     * 列表分页 返回数据
     *
     * @param list     数据数组
     * @param total    总数
     * @param order    序号 asc升序 desc降序
     * @param limit    每页条数
     * @param currPage 页码 从0开始
     * @return
     */
    public static RTableData okByEntity(List<?> list, Integer total, String order, Integer limit, Integer currPage) {
        if (list != null && list.size() > 0) {
            if (StringUtils.isNotBlank(order) && limit != null && currPage != null && currPage >= 0) {
                if (limit <= 0) {
                    //查询全部
                    limit = total;
                }
                for (int i = 0; i < list.size(); i++) {
                    //自动生成序号 字段:iid
                    ((ResponseBaseTableEntity) list.get(i)).setIid("asc".equalsIgnoreCase(order) ? limit * currPage + i + 1 : total - (limit * currPage) - i);
                }
            }
        }
        RTableData rspData = new RTableData();
        rspData.setList(list == null || list.size() == 0 ? new ArrayList<>() : list);
        rspData.setTotal(total);

        return rspData;
    }

    /**
     * 列表分页 返回数据 Map<String, Object>
     *
     * @param list     数据数组
     * @param total    总数
     * @param order    序号 asc升序 desc降序
     * @param limit    每页条数
     * @param currPage 页码 从0开始
     * @return
     */
    public static RTableData okByMap(List<?> list, Integer total, String order, Integer limit, Integer currPage) {
        if (list != null && list.size() > 0) {
            if (StringUtils.isNotBlank(order) && limit != null && currPage != null && currPage >= 0) {
                if (limit <= 0) {
                    //查询全部
                    limit = total;
                }
                for (int i = 0; i < list.size(); i++) {
                    //自动生成序号 字段:iid
                    ((Map<String, Object>) list.get(i)).put("iid", ("asc".equalsIgnoreCase(order) ? limit * currPage + i + 1 : total - (limit * currPage) - i));
                }
            }
        }
        RTableData rspData = new RTableData();
        rspData.setList(list == null || list.size() == 0 ? new ArrayList<>() : list);
        rspData.setTotal(total);

        return rspData;
    }
}
